Journals
  Publication Years
  Keywords
Search within results Open Search
Please wait a minute...
For Selected: Toggle Thumbnails
Lung tumor image recognition algorithm based on cuckoo search and deep belief network
YANG Jian, ZHOU Tao, GUO Lifang, ZHANG Feifei, LIANG Mengmeng
Journal of Computer Applications    2018, 38 (11): 3225-3230.   DOI: 10.11772/j.issn.1001-9081.2018041244
Abstract407)      PDF (957KB)(313)       Save
Due to random initialization of the weights, Deep Belief Network (DBN) easily falls into a local optimum, the Cuckoo Search (CS) algorithm was introduced into the traditional DBN model and a lung cancer image recognition algorithm based on CS-DBN was proposed. Firstly, the global optimization ability of CS was used to optimize initial weights of DBN, and on this basis, the layer-by-layer pre-training of DBN was performed. Secondly, the whole network was fine-tuned by using Back Propagation (BP) algorithm, so that the network weights were optimized. Finally, the CS-DBN was applied to the identification of lung tumor images, and CS-DBN was compared with traditional DBN from the four perspectives of Restricted Boltzmann Machine (RBM) training times, training batch sizes, DBN hidden layers numbers, and hidden layer nodes to verify the feasibility and effectiveness of the algorithm. The experimental results show that the recognition accuracy of CS-DBN is obviously higher than that of traditional DBN. Under the conditions of different RBM training times, training batch sizes, DBN hidden layer numbers, and hidden layer nodes, the increase range of CS-DBN identification accuracy over traditional DBN are 1.13 to 4.33, 2 to 3.34, 1.07 to 3.34 and 1.4 to 3.34 percentage points respectively. CS-DBN can improve the accuracy of lung tumor recognition to a certain extent, thereby improving the performance of computer-aided diagnosis of lung tumors.
Reference | Related Articles | Metrics